# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false %s -tf-graph-as-function -o - | FileCheck %s

node {
  name: "arg"
  op: "_Arg"
  attr {
    key: "T"
    value {
      type: DT_INT32
    }
  }
  attr {
    key: "index"
    value {
      i: 0
    }
  }
}
node {
  name: "ret"
  op: "_Retval"
  input: "arg"
  attr {
    key: "T"
    value {
      type: DT_INT32
    }
  }
  attr {
    key: "index"
    value {
      i: 0
    }
  }
}
versions {
  producer: 27
}

# CHECK:      func @main(%[[ARG_0:[a-z0-9]+]]: tensor<*xi32>) -> tensor<*xi32>
# CHECK-SAME: control_outputs = ""
# CHECK-SAME: inputs = "arg"
# CHECK-SAME: outputs = "ret"
# CHECK:        %[[GRAPH:[0-9]+]] = tf_executor.graph
# CHECK:          tf_executor.fetch %[[ARG_0]]
# CHECK:        return %[[GRAPH]]
